package com.test;

import java.util.Random;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

/**
 * https://zhuanlan.zhihu.com/p/148231820
 *
 *
 *    Java的concurrent包里面的CountDownLatch其实可以把它看作一个计数器，
 *    只不过这个计数器的操作是原子操作，同时只能有一个线程去操作这个计数器，
 *    也就是同时只能有一个线程去减这个计数器里面的值。
 *
 *       你可以向CountDownLatch对象设置一个初始的数字作为计数值，
 *       任何调用这个对象上的await()方法都会阻塞，直到这个计数器的计数值被其他的线程减为0为止。
 *
 *       CountDownLatch的一个非常典型的应用场景是：
 *       有一个任务想要往下执行，但必须要等到其他的任务执行完毕后才可以继续往下执行。
 *       假如我们这个想要继续往下执行的任务调用一个CountDownLatch对象的await()方法，
 *       其他的任务执行完自己的任务后调用同一个CountDownLatch对象上的countDown()方法，
 *       这个调用await()方法的任务将一直阻塞等待，直到这个CountDownLatch对象的计数值减到0为止
 */
public class CountDownLatchDemo {

    public static void main(String[] args) {
        ExecutorService executor = Executors.newCachedThreadPool();

        CountDownLatch latch = new CountDownLatch(3);

        Worker w1 = new Worker(latch,"张三");
        Worker w2 = new Worker(latch,"李四");
        Worker w3 = new Worker(latch,"王二");

        Boss boss = new Boss(latch);

        executor.execute(w3);
        executor.execute(w2);
        executor.execute(w1);
        executor.execute(boss);

        executor.shutdown();
    }

}

class Boss implements Runnable {

    private CountDownLatch downLatch;

    public Boss(CountDownLatch downLatch){
        this.downLatch = downLatch;
    }

    public void run() {
        System.out.println("老板正在等所有的工人干完活......");
        try {
            this.downLatch.await();
        } catch (InterruptedException e) {
        }
        System.out.println("工人活都干完了，老板开始检查了！");
    }

}

class Worker implements Runnable{

    private CountDownLatch downLatch;
    private String name;

    public Worker(CountDownLatch downLatch, String name){
        this.downLatch = downLatch;
        this.name = name;
    }

    public void run() {
        this.doWork();
        try{
            TimeUnit.SECONDS.sleep(new Random().nextInt(10));
        }catch(InterruptedException ie){
        }
        System.out.println(this.name + "活干完了！");
        this.downLatch.countDown();

    }

    private void doWork(){
        System.out.println(this.name + "正在干活!");
    }

}